package cn.hfatec.shms.web.jg.jgGroup.service;

import cn.hfatec.Page;
import cn.hfatec.Table;
import cn.hfatec.base.BaseDao;
import cn.hfatec.shms.advice.annotation.MysqlDataSourceXm;
import cn.hfatec.shms.cache.CodeInfoCache;
import cn.hfatec.shms.enums.CodeType;
import cn.hfatec.shms.system.codeInfo.model.CodeInfo;
import cn.hfatec.shms.web.WeekTime;
import cn.hfatec.shms.web.jg.jgGroup.dao.JgGroupDao;
import cn.hfatec.shms.web.jg.jgGroup.model.JgGroup;
import cn.hfatec.shms.web.jg.jgGroup.model.JgGroupParams;
import org.apache.poi.hssf.usermodel.*;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;


/**
 * @author hfatec
 * @version 1.0
 *          <p/>
 *          the JAVA code is generate by middlegen
 */
@Service("JgGroupService")
public class JgGroupService {

    public static final String BEAN_ID = "JgGroupService";

    //注入Dao层
    @Resource(name = "JgGroupDao")
    private JgGroupDao jgGroupDao;

    //注入BaseDao
    @Resource(name = "BaseDao")
    private BaseDao baseDao;

    @Transactional
    @MysqlDataSourceXm
    public static String encodeFilename(HttpServletRequest request, String fileName) {
        String rtn = "";
        try {
            String userAgent = request.getHeader("USER-AGENT");
            String new_filename = URLEncoder.encode(fileName, "UTF8");

            rtn = "filename=\"" + new_filename + "\"";

            if (userAgent != null) {
                userAgent = userAgent.toLowerCase();

                if (userAgent.indexOf("msie") != -1) {
                    rtn = "filename=\"" + new_filename + "\"";
                } else if (userAgent.indexOf("opera") != -1) {
                    rtn = "filename*=UTF-8''" + new_filename;
                } else if (userAgent.indexOf("safari") != -1) {
                    rtn = "filename=\"" + new String(fileName.getBytes("UTF-8"), "ISO8859-1") + "\"";
                } else if (userAgent.indexOf("applewebkit") != -1) {
                    new_filename = new String(fileName.getBytes("UTF-8"), "ISO-8859-1");
                    rtn = "filename=\"" + new_filename + "\"";
                } else if (userAgent.indexOf("mozilla") != -1) {
                    rtn = "filename*=UTF-8''" + new_filename;
                }
            }
        } catch (UnsupportedEncodingException e) {
            try {
                rtn = new String(fileName.getBytes("UTF-8"), "iso-8859-1");
            } catch (UnsupportedEncodingException localUnsupportedEncodingException1) {
            }
        }
        return rtn;
    }

    /**
     * 查询数据
     *
     * @param jgGroupParams
     * @param page
     */
    @Transactional
    @MysqlDataSourceXm
    public Table queryJgGroupTable(JgGroupParams jgGroupParams, Page page) {

        HashMap<String, Object> params = new HashMap<String, Object>();
        if (jgGroupParams.getStartTime() == null && jgGroupParams.getEndTime() == null) {

            jgGroupParams.setStartTime(WeekTime.getMonthFirstDay());
            jgGroupParams.setEndTime(WeekTime.getMonthLastDay());
        }
        params.put("params", jgGroupParams);
        return baseDao.queryTableByPage("queryJgGroup", params, page);
    }

    /**
     * 插入数据
     *
     * @param jgGroup
     */
    @Transactional
    @MysqlDataSourceXm
    public void saveJgGroup(JgGroup jgGroup) {
        if (jgGroup.getId() != null) {
            JgGroup result = getJgGroup(jgGroup.getId());
            if (result != null) {
                jgGroupDao.update(jgGroup);
            } else {
                jgGroupDao.insert(jgGroup);
            }
        } else {
            jgGroupDao.insert(jgGroup);
        }
    }

    /**
     * 删除数据
     *
     * @param id
     */
    @Transactional
    @MysqlDataSourceXm
    public void deleteJgGroup(Integer id) {

        jgGroupDao.delete(id);
    }

    /**
     * 删除数据
     *
     * @param jgGroupList
     */
    @Transactional
    @MysqlDataSourceXm
    public void deleteJgGroupBatch(List<String> jgGroupList) {
        for (String id : jgGroupList) {
            deleteJgGroup(Integer.valueOf(id));
        }
    }

    /**
     * 根据ID获取数据
     *
     * @param id
     */
    @Transactional
    @MysqlDataSourceXm
    public JgGroup getJgGroup(Integer id) {

        return jgGroupDao.getJgGroupById(id);
    }

    /**
     * 导出
     *
     * @param
     */
    @Transactional
    @MysqlDataSourceXm
    public String exportExcel(JgGroupParams jgGroupParams, HttpServletRequest request, HttpServletResponse response) {
        try {
            String clientFileName = encodeFilename(request, System.currentTimeMillis() + "");
            response.reset();
            response.setHeader("Content-Disposition", "attachment; " + clientFileName + ".xls");
            response.setContentType("application/octet-stream;charset=UTF-8");
            OutputStream os = response.getOutputStream();
            List dataList = jgGroupDao.queryJgGroup(jgGroupParams);
            List nameList = new ArrayList();
            Calendar c = Calendar.getInstance();
            int weeknum = c.get(Calendar.WEEK_OF_YEAR);
            nameList.add("三师负责人");

            nameList.add("二级管控");
            nameList.add("二级管控客户数");
            nameList.add("三级管控");
            nameList.add("三级管控客户数");

            nameList.add("保有量");
            this.exportExcel(nameList, dataList, os);
        } catch (Exception e) {
            e.printStackTrace();
        }

        return "";
    }

    /**
     * @return HSSFWorkbook
     */
    public HSSFWorkbook createWorkbook(List fieldName, List fieldData) {
        List<CodeInfo> sexList = CodeInfoCache.getCodeInfo(CodeType.SEX);
        // 第一步，创建一个webbook，对应一个Excel文件
        HSSFWorkbook wb = new HSSFWorkbook();
        // 第二步，在webbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet("分组汇总");
        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
        HSSFRow row = sheet.createRow((int) 0);
        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        HSSFCellStyle cellstyle = (HSSFCellStyle) wb.createCellStyle();// 设置表头样式
        cellstyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER); // 设置垂直居中
        cellstyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);// 设置水平居中
        cellstyle.setWrapText(true); // 设置为自动换行
        HSSFFont cellFont = (HSSFFont) wb.createFont(); // 创建表头字体样式
        cellFont.setFontHeightInPoints((short) 10); // 设置字体大小
        cellFont.setFontName("宋体"); // 设置字体类型
        cellFont.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD); // 字体加粗
        cellstyle.setFont(cellFont); // 为表头样式设置字体样式


        for (int i = 0; i < fieldName.size(); i++) {
            HSSFCell cell = row.createCell((short) i);
            cell.setCellValue((String) fieldName.get(i));
            cell.setCellStyle(cellstyle);
        }

        JgGroup jgGroup = null;

        // 第五步，写入实体数据 实际应用中这些数据从数据库得到，

        for (int i = 0; i < fieldData.size(); i++) {
            row = sheet.createRow(i + 1);
            jgGroup = (JgGroup) fieldData.get(i);
            // 第四步，创建单元格，并设置值
            String[] nameArry = {jgGroup.getResponsibleName(),
                    jgGroup.getTwoControl(), jgGroup.getTwoCustom(),
                    jgGroup.getThreeControl(),
                    jgGroup.getThreeCustom(), jgGroup.getRetainNum()
            };
            for (int j = 0; j < nameArry.length; j++) {
                if (nameArry[j] == null) {
                    row.createCell((short) j).setCellValue("");
                } else {
                    row.createCell((short) j).setCellValue(nameArry[j]);
                }

            }
        }
        //自适应列宽
        for (int i = 0; i < fieldName.size(); i++) {
            sheet.autoSizeColumn((short) i);
        }
        return wb;
    }

    //将信息写入输出流的方法。
    @MysqlDataSourceXm
    public void exportExcel(List fieldName, List fieldData, OutputStream os) throws Exception {
        HSSFWorkbook wb = createWorkbook(fieldName, fieldData);
        wb.write(os);
        os.close();
    }
}